package com.example.lib_mvp.views;

import android.content.Context;
import android.graphics.Canvas;
import android.graphics.Color;
import android.graphics.Paint;
import android.util.AttributeSet;
import android.view.View;

public class AudioVisualizerView extends View {

    private Paint paint; // 用于绘制的画笔
    private int[] amplitudes; // 存储音频幅度

    public AudioVisualizerView(Context context, AttributeSet attrs) {
        super(context, attrs);
        // 初始化画笔
        paint = new Paint();
        paint.setColor(Color.GREEN); // 设置颜色为绿色
        paint.setStrokeWidth(2); // 设置线宽
        amplitudes = new int[30]; // 假设我们需要显示30根柱子
    }

    @Override
    protected void onDraw(Canvas canvas) {
        super.onDraw(canvas);
        // 绘制音频柱子
        drawBars(canvas);
    }

    // 绘制音频柱子的方法
    private void drawBars(Canvas canvas) {
        int width = getWidth();
        int height = getHeight();
        int barWidth = width / amplitudes.length; // 每根柱子的宽度

        for (int i = 0; i < amplitudes.length; i++) {
            // 根据幅度绘制柱子的高度
            int barHeight = (int) ((amplitudes[i] / 255.0) * height);
            // 在画布上绘制柱子
            canvas.drawLine(i * barWidth, height, i * barWidth, height - barHeight, paint);
        }
    }

    // 设置音频幅度的方法
    public void setAmplitudes(int[] amplitudes) {
        this.amplitudes = amplitudes;
        // 重新绘制
        invalidate();
    }

    public int[] setAmplitudes(short[] buffer) {
        int[] amplitudes = new int[30];
        for (int i = 0; i < amplitudes.length; i++) {
            // 此处取平均振幅
            amplitudes[i] = Math.abs(buffer[i * (buffer.length / amplitudes.length)]);
        }
        return amplitudes;
    }
}